package TextRank;

import dataStructure.ArffFuntion;
import weka.core.Instances;
import weka.core.converters.ConverterUtils;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;


public class DealArff {

    public List<ArffFuntion> DealArff(String loadpath) throws Exception {
        //读取arff文件，并获取数据
        Instances data = ConverterUtils.DataSource.read(loadpath);   //下标从0开始
        List<ArffFuntion> list = new ArrayList<>();                  //用来存放arff中的数据

        List<String> list2 = new ArrayList<>();
        //遍历文件中内容并存储下来
        for (int i = 0; i < data.numInstances(); i++) {
            ArffFuntion ar = new ArffFuntion();
            String [] str1 = data.get(i).toString(0).split("\\.");         //获取第一列数据
            String temp = data.get(i).toString(1).replace("'","");   //去除字符“ '”
            list2 = Arrays.asList(temp.split(" "));         //获取第二列数据
            ar.setClassname(str1[str1.length-2]);
            ar.setMethodname(str1[str1.length-1]);
            ar.setList(list2);
            list.add(ar);
        }
        return list;
        //todo将netty.arff文件中的信息存储起来
    }

}
